import { Module } from '@nestjs/common'
import { ConfigModule } from '@nestjs/config'
import { MongooseModule } from '@nestjs/mongoose'
import { ComfyService } from './comfy.service'
import { WorkflowBuilder } from './workflow.builder'
import { WorkflowTemplateManager } from './workflow.template'
import { WorkflowTemplate, WorkflowTemplateSchema } from './schemas/workflow-template.schema'

@Module({
    imports: [
        ConfigModule,
        MongooseModule.forFeature([
            { name: WorkflowTemplate.name, schema: WorkflowTemplateSchema }
        ])
    ],
    providers: [
        ComfyService,
        WorkflowBuilder,
        WorkflowTemplateManager
    ],
    exports: [ComfyService, WorkflowTemplateManager]
})
export class ComfyModule { } 